<template>
  <div class="login-container">
    <el-card class="login-card">
      <div slot="header" class="card-header">
        <h2>登录</h2>
      </div>
      <el-form 
        ref="loginForm" 
        :model="loginForm" 
        :rules="rules" 
        label-width="80px">
        <el-form-item label="身份" prop="role">
          <el-select 
            v-model="loginForm.role" 
            placeholder="请选择身份"
            @change="handleRoleChange">
            <el-option label="管理员" value="admin"></el-option>
            <el-option label="教师" value="teacher"></el-option>
            <el-option label="学生" value="student"></el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="用户号" prop="userId">
          <el-input 
            v-model="loginForm.userId"
            placeholder="请输入用户号">
          </el-input>
        </el-form-item>
        <el-form-item label="密码" prop="password">
          <el-input 
            v-model="loginForm.password"
            type="password"
            placeholder="请输入密码"
            @keyup.enter.native="handleLogin">
          </el-input>
        </el-form-item>
        <el-form-item>
          <el-button 
            type="primary" 
            :loading="loading"
            @click="handleLogin"
            class="login-button">
            登录
          </el-button>
          <el-button 
            type="text"
            @click="$router.push('/admin/register')"
            class="register-link">
            还没有账号？立即注册
          </el-button>
        </el-form-item>
      </el-form>
    </el-card>
  </div>
</template>

<script>
import { userApi } from '@/api'

export default {
  name: 'Login',
  data() {
    // 用户号格式验证
    const validateUserId = (rule, value, callback) => {
      if (!value) {
        return callback(new Error('请输入用户号'))
      }
      if (!this.loginForm.role) {
        return callback(new Error('请先选择身份'))
      }
      
      const isNumber = /^\d+$/.test(value)
      if (!isNumber) {
        return callback(new Error('用户号必须是纯数字'))
      }

      const length = value.length
      switch (this.loginForm.role) {
        case 'admin':
          if (length < 1 || length > 5) {
            callback(new Error('管理员用户号必须是1-5位数字'))
          }
          break
        case 'teacher':
          if (length < 6 || length > 9) {
            callback(new Error('教师用户号必须是6-9位数字'))
          }
          break
        case 'student':
          if (length < 10) {
            callback(new Error('学生用户号必须是10位及以上数字'))
          }
          break
        default:
          callback(new Error('请先选择正确的身份'))
      }
      callback()
    }

    return {
      loginForm: {
        userId: '',
        password: '',
        role: ''
      },
      rules: {
        role: [
          { required: true, message: '请选择身份', trigger: 'change' }
        ],
        userId: [
          { required: true, validator: validateUserId, trigger: 'blur' }
        ],
        password: [
          { required: true, message: '请输入密码', trigger: 'blur' },
          { min: 6, max: 20, message: '密码长度必须在6-20个字符之间', trigger: 'blur' }
        ]
      },
      loading: false
    }
  },
  methods: {
    // 身份改变时重新验证用户号
    handleRoleChange() {
      if (this.loginForm.userId) {
        this.$refs.loginForm.validateField('userId')
      }
    },
    async handleLogin() {
      try {
        await this.$refs.loginForm.validate()
        this.loading = true
        
        // 直接调用登录接口
        try {
          const res = await userApi.login(this.loginForm)
          if (res.code === 200) {
            // 保存用户信息
            localStorage.setItem('user', JSON.stringify(res.data))
            this.$message.success('登录成功')
            
            // 根据角色跳转到不同页面
            switch (res.data.role) {
              case 'admin':
                // 管理员直接进入用户管理页面
                this.$router.push('/admin/users')
                break
              case 'teacher':
                // 教师直接进入学生管理页面
                this.$router.push('/teacher/students')
                break
              case 'student':
                // 学生进入首页，可以开始测试
                this.$router.push('/home')
                break
              default:
                this.$router.push('/home')
            }
          } else {
            // 根据错误信息显示不同提示
            const errorMsg = res.msg || res.message || '登录失败'
            
            if (errorMsg.includes('用户号不存在')) {
              this.$message.error('用户号不存在，请先注册')
            } else if (errorMsg.includes('密码错误')) {
              this.$message.error('密码错误，请重试')
            } else if (errorMsg.includes('用户角色不匹配')) {
              this.$message.error(`该用户不是${this.getRoleName(this.loginForm.role)}角色`)
            } else {
              this.$message.error(errorMsg)
            }
          }
        } catch (error) {
          console.error('登录请求错误:', error)
          // 处理特定错误消息
          const errorMsg = error.msg || error.message || '登录失败，请检查输入信息'
          
          if (errorMsg.includes('用户号不存在')) {
            this.$message.error('用户号不存在，请先注册')
          } else if (errorMsg.includes('密码错误')) {
            this.$message.error('密码错误，请重试')
          } else if (errorMsg.includes('用户角色不匹配')) {
            this.$message.error(`该用户不是${this.getRoleName(this.loginForm.role)}角色`)
          } else {
            this.$message.error(errorMsg)
          }
        }
      } catch (error) {
        console.error('表单验证错误：', error)
        this.$message.error('请正确填写登录信息')
      } finally {
        this.loading = false
      }
    },
    getRoleName(role) {
      switch (role) {
        case 'admin': return '管理员'
        case 'teacher': return '教师'
        case 'student': return '学生'
        default: return '未知'
      }
    }
  }
}
</script>

<style scoped>
.login-container {
  display: flex;
  justify-content: center;
  align-items: center;
  min-height: 100vh;
  background-color: #f5f7fa;
}

.login-card {
  width: 400px;
  border-radius: 8px;
}

.card-header {
  text-align: center;
  margin-bottom: 20px;
}

.card-header h2 {
  margin: 0;
  color: #303133;
}

.login-button {
  width: 100%;
  margin-bottom: 10px;
}

.register-link {
  display: block;
  text-align: center;
  margin-top: 10px;
}

.el-select {
  width: 100%;
}
</style>